Tangram: Unlocking Non-Uniform KV Cache Compression for Efficient Multi-turn LLM Serving
解锁非均匀KV缓存压缩以高效支持多轮LLM服务
🧭 LLM服务 · 非均匀KV缓存压缩
Hanyang University · Rebellions
提出Tangram框架,通过静态预算预留、粗糙分页和提前负载均衡,使非均匀KV压缩在多轮LLM服务中可达最高2.6倍吞吐提升
📌 工作参考价值(4 点)
- 多轮对话推荐 / Agent 场景中,KV 缓存膨胀是吞吐瓶颈,非均匀头级压缩精度保持更好,Tangram 展示了如何通过离线校准头的保留模式并静态规划调度,避免运行时碎片回收开销,对自研推理引擎有直接参考价值。
- 可借鉴“预算预留”:请求调度时预先固定每头压缩后内存大小,消除 prefill 阶段的页面回收(原占 25% prefill 时间),减少延迟抖动,适合对延迟敏感的在线服务。
- “粗糙分页”将预算相近的头分入独立页表,将内部碎片转化为可回收内存,提升显存利用率,利于在有限显存下支撑更长对话历史。
- “提前负载均衡”离线计算 GPU 分区,彻底消除 decode 步中 15-20% 的动态重规划开销,这一思想可迁移到需要 GPU 并行推理的推荐模型部署中。